Add XPU support for AOT inductor #1503
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchchat/1503
Note: Links to docs will display an error until the docs builds have been completed. ⏳ 6 Pending, 1 Unrelated FailureAs of commit 6bece4f with merge base d3cd165 ( FLAKY - The following job failed but was likely due to flakiness present on trunk:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
facebook-github-bot
added
the
CLA Signed
| if [[ -x "$(command -v xpu-smi)" ]]; | ||
| then | ||
| ( | ||
| set -x |
There was a problem hiding this comment.
Can we reuse the conditional on line 84?
There was a problem hiding this comment.
No, because line 84 is to install using pytorch xpu link: https://download.pytorch.org/whl/nightly/xpu. Since we don't have torchtune nightly for xpu now, we temporarily to install torchtune cpu nightly from cpu link: https://download.pytorch.org/whl/nightly/cpu. After xpu nightly is ready, we'll update line 84 to install from pytorch xpu link.
There was a problem hiding this comment.
Ah I meant reuse the if [[ -x "$(command -v xpu-smi)" ]]; check
This PR is fine to merge as-is though since after we get torchtune nightly this is going to be removed anyways 😃
Add XPU support for AOT inductor.